// main.cc
#include "BlockQueue.hpp"
#include "Task.hpp"
#include <unistd.h>
#include <ctime>

void *Consumer(void *args)
{
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);

    while (true)
    {
        // 消费
        // sleep(1);
        Task t = bq->pop();

        // 处理数据
        // t.run();
        t();
        usleep(10000); // 主要是为了防止打印错乱
        std::cout << "处理任务：" << t.GetResult() << ", thread id: " << (int*)pthread_self() << std::endl;
    }
}

void *Productor(void *args)
{
    BlockQueue<Task> *bq = static_cast<BlockQueue<Task> *>(args);
    int len = opers.size();
    while (true)
    {
        // 模拟生产者生产数据
        int data1 = rand() % 10; // [0, 9]
        int data2 = rand() % 10; // [0, 9]
        char op = opers[rand() % len];
        Task t(data1, data2, op);
        // usleep(10);

        // 生产
        sleep(1);
        bq->push(t);
        std::cout << "生产了一个任务: " << t.GetTask() << ", thread id: " << (int*)pthread_self() << std::endl;
    }
}

int main()
{
    srand(time(nullptr));
    BlockQueue<Task> *bq = new BlockQueue<Task>;
    pthread_t c[3], p[5];
    for (int i = 0; i < 3; ++i)
    {
        pthread_create(c + i, nullptr, Consumer, bq);
    }
    for (int i = 0; i < 5; ++i)
    {
        pthread_create(p + i, nullptr, Productor, bq);
    }

    for (int i = 0; i < 3; ++i)
    {
        pthread_join(c[i], nullptr);
    }
    for (int i = 0; i < 5; ++i)
    {
        pthread_join(p[i], nullptr);
    }
    delete bq;
    return 0;
}